home *** CD-ROM | disk | FTP | other *** search
/ 9-Digit Zip Code Directory / 9-Digit Zip Code Directory (American Business Information) (ABIZIP-12).ISO / z4src.zip / BSUMB.C < prev    next >
C/C++ Source or Header  |  1995-04-26  |  4KB  |  166 lines
  1. //----------------------------------------------------------------------------
  2. //                            MODULE DESCRIPTION
  3. //
  4. //  Module:    bsumb.c
  5. //   Title:    Base library
  6. //  Notice:    John M. Weeder
  7. //                 Copyright (c) 1993. All rights reserved.
  8. //             This module contains proprietary information and should be 
  9. //                treated as confidential.
  10. //
  11. //----------------------------------------------------------------------------
  12. //                           MAINTENANCE HISTORY
  13. //
  14. // $Workfile$
  15. // $Revision$
  16. //   $Author$
  17. //     $Date$
  18. //      $Log$    
  19. //
  20. //----------------------------------------------------------------------------
  21. //                             MODULE NARRATIVE
  22. //
  23. //
  24. //    This module contains to manage DOS UMBs.
  25. //
  26. //    The code in this module should be written entirely in C. 
  27. //    Do not use any C++ constructs.
  28. //
  29. //    This module is portable to:
  30. //        DOS 3.X+
  31. //        MS Windows 3.X+
  32. //        OS/2 2.X+
  33. //        OS/2 2.0 PM
  34. //        SCO UNIX.
  35. //
  36. //    The following compilers are supported:
  37. //        MSC 6.0A
  38. //        MSC/C++ 7.0
  39. //        Borland C++ 3.1 for DOS
  40. //        Borland C++ 1.0 for OS/2 2.X
  41. //        SCO UNIX cc
  42. //
  43. //----------------------------------------------------------------------------
  44. #include <bs.h>
  45.  
  46.  
  47. //----------------------------------------------------------------------------
  48. //    Globals
  49. //----------------------------------------------------------------------------
  50. //#if OS_DOS
  51. static BOOL fInit = FALSE;
  52. static int sUmbSave;                            // Save initial state
  53. //#endif
  54.  
  55. //----------------------------------------------------------------------------
  56. //   Description:    Get umb state
  57. //    Parameters:
  58. //       Returns:    UMB state
  59. //----------------------------------------------------------------------------
  60. SHORT FN_E UmbGetState(void)
  61. {
  62. #if OS_DOS
  63.     SHORT sUmb;
  64.     if (_osmajor < 5 || (_osmajor >= 10 && _osmajor < 20))
  65.         return 0;
  66.     _asm {
  67.         mov ax, 5802h
  68.         int 21h
  69.         xor ah, ah
  70.         mov [sUmb], ax
  71.         }
  72.     return sUmb;
  73. #else
  74.     return 0;
  75. #endif
  76. }
  77.  
  78.  
  79. //----------------------------------------------------------------------------
  80. //   Description:    
  81. //    Parameters:    
  82. //       Returns:
  83. //----------------------------------------------------------------------------
  84. VOID FN_E UmbRestore(void)
  85. {
  86.     if (fInit)
  87.         {
  88.         UmbSetState(sUmbSave);
  89.         fInit = FALSE;
  90.         }
  91.     return ;
  92. }
  93.  
  94.  
  95. //----------------------------------------------------------------------------
  96. //   Description:    Set umb state
  97. //    Parameters:    sUmb        UMB state.
  98. //                                    1 = Use UMBs
  99. //                                    0 = Don't use UMBs
  100. //       Returns:    
  101. //----------------------------------------------------------------------------
  102. VOID FN_E UmbSetState(SHORT sUmb)
  103. {
  104. #if OS_DOS
  105.     USHORT usMajor;
  106.  
  107.     Version(&usMajor, NULL);
  108.  
  109.     if (_osmajor < 5 || (_osmajor >= 10 && _osmajor < 20))
  110.         return ;
  111.  
  112.     if (EnvCheck("NO_UMB", "YES"))
  113.         return ;
  114.  
  115.     if (!fInit)
  116.         {
  117.         sUmbSave = UmbGetState();
  118.         fInit = TRUE;
  119.         BaseExitFunc((PFNEXIT)UmbRestore, SYS_EXIT_PRIORITY);
  120.         }
  121.     sUmb = sUmb ? 1: 0;
  122.     _asm {
  123.         mov ax, 5803h
  124.         mov bx, [sUmb]
  125.         int 21h
  126.         }
  127.     return ;
  128. #else
  129.     NOTUSED(sUmb);
  130.     return ;
  131. #endif
  132. }
  133.  
  134.  
  135. //----------------------------------------------------------------------------
  136. //   Description:    Run standard test suite
  137. //    Parameters:    sTest        Test to run.
  138. //                                        0        Run all default tests (except).
  139. //       Returns:    TRUE if successful.
  140. //----------------------------------------------------------------------------
  141. #if COMPILE_TEST
  142. BOOL FN UmbTest(SHORT sTest)
  143. {
  144.     LONG lFree;
  145.     LONG lProgram;
  146.     
  147.     NOTUSED(sTest);
  148.  
  149.     Output("UMB enabled.\n");
  150.     UmbSetState(1);
  151.     ProgramSize(&lProgram, &lFree);
  152.     Output("%ld bytes used.\n%ld bytes free. %ld total.\n",
  153.         lProgram, lFree, lProgram+lFree);
  154.  
  155.     Output("UMB disabled.\n");
  156.     UmbSetState(0);
  157.     ProgramSize(&lProgram, &lFree);
  158.     Output("%ld bytes used.\n%ld bytes free. %ld total.\n",
  159.         lProgram, lFree, lProgram+lFree);
  160.     return TRUE;
  161. }
  162. #endif
  163. //----------------------------------------------------------------------------
  164. //------------------------------- End of File --------------------------------
  165. //----------------------------------------------------------------------------
  166.